# 数据预处理
import pandas as pd
data_pretreatment = pd.read_csv("data/order.csv")
data_pretreatment.drop(['customer','order','hour','weekday','total_items'],axis = 1,inplace=True) 
print(data_pretreatment)
# 将数据保存
data_pretreatment.to_csv("tmp/KMeans_training_data.csv",index=False)

# 保留customer列
df = pd.read_csv("data/order.csv")
df.drop(['order','total_items','weekday','hour'],axis=1,inplace=True)
# 数据保存
df.to_csv("tmp/hierarchy_training_data.csv",index=False)
print(df)